import java.util.ArrayList;
import java.util.LinkedList;

/**
 * 给定一个由 0 和 1 组成的矩阵 mat ，请输出一个大小相同的矩阵，其中每一个格子是 mat 中对应位置元素到最近的 0 的距离。
 * <p>
 * 两个相邻元素间的距离为 1 。
 */
public class Solution542 {

    int[][] dirs = {{0, 1}, {1, 0}, {-1, 0}, {0, -1}};
    public int[][] updateMatrix(int[][] mat) {
        //要什么我new什么
        int m = mat.length;//多少行
        int n = mat[0].length;//多少列
        int[][] res = new int[m][n];

        //用队列存储这一层的数据。用矩阵存储访问的情况。
        boolean[][] visited = new boolean[m][n];
        LinkedList<int[]> queue = new LinkedList<>();//int[] 用于存储点的坐标 [1,2]

        //1. 首先将0放入矩阵中。
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (mat[i][j] == 0) {
                    queue.offer(new int[]{i, j});//把这些点保存到队列中
//                    res[i][i] = 0;//自己到自己的距离为0，这个地方不需要初始化，因为默认值都为0
                    visited[i][j] = true;
                }
            }
        }

        //2.当队列中有数据的时候就，循环这一层，
        int level = 0;
        while (!queue.isEmpty()) {
            //2.1 循环这一层每一个数
            int size = queue.size();
            for (int k = 0; k < size; k++) {
                int[] cur = queue.poll();
                int i = cur[0];
                int j = cur[1];
                //mat只有0和1，这里的1是0后下一层的1
                if (mat[i][j] == 1) {
                    res[i][j] = level;
                }
                //2.2 将这一层每一个、相邻的、没有访问的点加入到队列中
                for (int[] dir : dirs) {
                    int x = i + dir[0];
                    int y = j + dir[1];
                    if (x >= 0 && x < m && y >= 0 && y < n && !visited[x][y]) {
                        queue.offer(new int[]{x, y});
                        visited[x][y] = true;
                    }
                }
            }
            level++;
        }
        return res;
    }

    public static void main(String[] args) {
        int[][] a = {{0,0,0,0},{1,1,0,1},{1,0,1,1},{1,1,1,1},{1,1,1,1},{1,1,1,1}};
        int[][] ints = new Solution542().updateMatrix(a);
        for (int i = 0; i < ints.length; i++) {
            for (int j = 0; j < ints[0].length; j++) {
                System.out.print(ints[i][j]);
            }
            System.out.println();
        }
    }
}
